package com.pb.infra.oth.util;

import com.itextpdf.text.*;
import com.itextpdf.text.pdf.*;
import com.itextpdf.text.pdf.draw.LineSeparator;
import lombok.SneakyThrows;

import java.io.*;

/**
 * @auther: liaohaoyuan
 * @time: 2022/6/10 14:21
 * @description
 */
public class PdfHeaderFooter extends PdfPageEventHelper {
      /**
     * 文档字体大小，页脚页眉最好和文本大小一致
     */
    public int presentFontSize = 12;

    /**
     * logo  页眉
     */
    public Image image;

    // 模板
    public PdfTemplate total;

    // 基础字体对象
    public BaseFont bf = null;

    // 利用基础字体生成的字体对象，一般用于生成中文文字
    public Font fontDetail = null;

    public Font font = null;
    /**
     *
     * TODO 文档打开时创建模板
     *
     * @see PdfPageEventHelper#onOpenDocument(PdfWriter,
     *      Document)
     */
    @Override
    public void onOpenDocument(PdfWriter writer, Document document) {
        total = writer.getDirectContent().createTemplate(50, 50);// 共 页 的矩形的长宽高
    }

    /**
     *
     * TODO 关闭每页的时候，写入页眉，写入'第几页共'这几个字。
     *
     * @see PdfPageEventHelper#onEndPage(PdfWriter,
     *      Document)
     */
    @SneakyThrows
    @Override
    public void onEndPage(PdfWriter writer, Document document) {
        this.addPage(writer, document);
        //this.addWatermark(writer);
    }

    //加分页
    public void addPage(PdfWriter writer, Document document) throws BadElementException, IOException {
        //设置分页页眉页脚字体
        try {
            if (bf == null) {
                bf = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", false);
            }
            if (fontDetail == null) {
                fontDetail = new Font(bf, presentFontSize, Font.NORMAL);// 数据体字体
            }
            if (font == null) {
                font = new Font(bf, 8);// 数据体字体
            }
        } catch (DocumentException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }

        int pageS = writer.getPageNumber();
        String basePath = System.getProperty("user.dir")+"\\primefmapp-boot\\src\\main\\resources\\static\\1.png";
        InputStream inputStream = new FileInputStream(new File(basePath));
        int n;
        byte[] buffer = new byte[4096];
        ByteArrayOutputStream output = new ByteArrayOutputStream();
        while (-1 != (n = inputStream.read(buffer))) {
            output.write(buffer, 0, n);
        }
        // 页眉logo图片实例
        image = Image.getInstance(output.toByteArray());
        image.scalePercent(40);
        //  ！！！！ 最重要的是这个， 如果页眉需要设置图片的话，需要在Phrase对象中添加一个Chunk对象，在Chunk对象中添加图片信息即可
        Phrase p1 = new Phrase("", font);
        float center = document.getPageSize().getRight()/2-35;//页面的水平中点
         p1.add(new Chunk(image, 0, -30));
        // 1、写入左侧页眉logo
        ColumnText.showTextAligned(writer.getDirectContent(),
                Element.ALIGN_LEFT, p1,
                document.left(-35), document.top(-35), 0);

//             //1.写入页眉文字
//            ColumnText.showTextAligned(writer.getDirectContent(),
//                    Element.ALIGN_LEFT, new Phrase(header, fontDetail),
//                    document.left()-10, document.top() + 40, 0);
        // 1.写入页眉
//            // 直线
        this.addWatermark(writer);
        Paragraph p2 = new Paragraph();
        LineSeparator lineSeparator = new LineSeparator();
        lineSeparator.setLineColor(new BaseColor(180,180,180));
        lineSeparator.setLineWidth(0.3f);
        p2.add(new Chunk(lineSeparator));
        ColumnText.showTextAligned(writer.getDirectContent(),
                Element.ALIGN_LEFT, p2,
                document.left()-38, document.top() + 3, 0);

        // 2.写入前半部分的 第 X页/共
        String foot1 = "第 " + pageS + " 页 /共";
        Phrase footer = new Phrase(foot1, fontDetail);

        // 3.计算前半部分的foot1的长度，后面好定位最后一部分的'Y页'这俩字的x轴坐标，字体长度也要计算进去 = len
        float len = bf.getWidthPoint(foot1, presentFontSize);

        // 4.拿到当前的PdfContentByte
        PdfContentByte cb = writer.getDirectContent();

        // 5.写入页脚1，x轴就是(右margin+左margin + right() -left()- len)/2.0F
        // 再给偏移20F适合人类视觉感受，否则肉眼看上去就太偏左了
        // ,y轴就是底边界-20,否则就贴边重叠到数据体里了就不是页脚了；注意Y轴是从下往上累加的，最上方的Top值是大于Bottom好几百开外的。
        ColumnText
                .showTextAligned(
                        cb,
                        Element.ALIGN_CENTER,
                        footer,
                        (document.rightMargin() + document.right()
                                + document.leftMargin() - document.left() - len) / 2.0F + 20F,
                        document.bottom() - 20, 0);

        // 6.写入页脚2的模板（就是页脚的Y页这俩字）添加到文档中，计算模板的和Y轴,X=(右边界-左边界 - 前半部分的len值)/2.0F +
        // len ， y 轴和之前的保持一致，底边界-20 // 调节模版显示的位置
        cb.addTemplate(total, (document.rightMargin() + document.right()
                        + document.leftMargin() - document.left()) / 2.0F + 20F,
                document.bottom() - 20);



    }

    //加水印
    public void addWatermark(PdfWriter writer){
        // 水印图片
            // 设置透明度
            PdfGState gs = new PdfGState();
            gs.setFillOpacity(0.3f);
          PdfContentByte content = writer.getDirectContentUnder();
            content.beginText();
            // 开始写入水印

            // 水印的透明度
            content.setGState(gs);
            // 设置颜色 默认为蓝色
            content.setColorFill(BaseColor.BLACK);
            // content.setColorFill(Color.GRAY);
            // 设置字体及字号
            content.setFontAndSize(bf, 72);
            // 设置起始位置
            // content.setTextMatrix(400, 880);
            content.setTextMatrix(400, 880);
            // 开始写入水印
            content.showTextAligned(Element.ALIGN_LEFT, "内部文档，禁止外泄", 80, 380, 45);


            content.endText();

    }

    /**
     *
     * TODO 关闭文档时，替换模板，完成整个页眉页脚组件
     *
     * @see PdfPageEventHelper#onCloseDocument(PdfWriter,
     *      Document)
     */
    @Override
    public void onCloseDocument(PdfWriter writer, Document document) {
        // 7.最后一步了，就是关闭文档的时候，将模板替换成实际的 Y 值,至此，page x of y 制作完毕，完美兼容各种文档size。
        total.beginText();
        total.setFontAndSize(bf, presentFontSize);// 生成的模版的字体、颜色
        String foot2 = " " + writer.getPageNumber() + " 页"; //页脚内容拼接  如  第1页/共2页
        total.showText(foot2);// 模版显示的内容
        total.endText();
        total.closePath();
    }

}
